package cn.tedu.weibo.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class UploadController {

    @RequestMapping("/upload")
    public String upload(MultipartFile pic) throws IOException {
        System.out.println("pic" + pic);
        String fileName = pic.getOriginalFilename();
        System.out.println("原始文件名:" + fileName);
        String suffix = fileName.substring(fileName.lastIndexOf("."));
        fileName = UUID.randomUUID()+suffix;
        System.out.println("唯一文件名:"+fileName);

        String dirpath = "e:/temp";
        File dirFile = new File(dirpath);
        if (!dirFile.exists()){
            dirFile.mkdirs();
        }
        String filePath = dirpath+"/"+fileName;

        pic.transferTo(new File(filePath));

        return "/"+fileName;
    }

    @RequestMapping("/remove")
    public void remove(String url){
        // 删除磁盘上的文件
        new File("e:/temp"+url).delete();
    }
}
